import pymysql

# 导入配置文件
from views.settings import DB_NAME,DB_HOST,DB_USER,DB_PASSWORD

class Db:

	# 初始化方法
	def __init__(self):

		# 链接数据库
		self.conn = pymysql.connect(host=DB_HOST,user=DB_USER,password=DB_PASSWORD,db=DB_NAME,charset="utf8",cursorclass=pymysql.cursors.DictCursor)


	# 删除方法
	def delete(self,sql):

		# 创建游标对象
		self.cursor = self.conn.cursor()
		# 执行sql语句
		self.cursor.execute(sql)
		# 提交sql
		self.conn.commit()

		return "delete ok"



	# 插入方法
	def insert(self,sql):

		# 创建游标对象
		self.cursor = self.conn.cursor()
		# 执行sql语句
		self.cursor.execute(sql)
		# 提交sql
		self.conn.commit()

		return "insert ok"

	# 修改方法
	def update(self,sql):

		# 创建游标对象
		self.cursor = self.conn.cursor()
		# 执行sql语句
		self.cursor.execute(sql)
		# 提交sql
		self.conn.commit()

		return "update ok"

	# 查询方法 单个数据
	def get_one(self,sql):

		# 创建游标对象
		self.cursor = self.conn.cursor()
		# 执行sql语句
		self.cursor.execute(sql)
		# 获取数据
		res = self.cursor.fetchone()
		# 返回结果集
		return res

	# 查询方法 多个数据
	def get_all(self,sql):

		# 创建游标对象
		self.cursor = self.conn.cursor()
		# 执行sql语句
		self.cursor.execute(sql)
		# 获取数据
		res = self.cursor.fetchall()
		# 返回结果集
		return res


	# 析构方法
	def __del__(self):

		# 关闭游标
		self.cursor.close()
		# 关闭数据库链接
		self.conn.close()

db = Db()

# res = db.get_all(" select * from dept ")
# res = db.get_all(" select * from test ")

# print(res)